package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 聊天室的服务端 版本5
 * 完善服务端的功能
 * 1.添加广播某个客户端上下线的情况(并且广播当前在线人数)
 *
 * 实现:
 *
 */

public class Server5 {
    /*
    java.net.ServerSocket
    运行在服务端的ServerSocket的主要任务:
    1.向服务端申请服务端口,客户端就是通过改端口与服务器建立连接的
    2.监听服务端口,一旦一个客户端连接了就会立即创建一个Socket实例与之交互
    我们将ServerSocket比喻为某客服中心的"总机"
     */
    private List<PrintWriter> allOut = new ArrayList<>();
    //V4:该集合用于保存所有客户端的输出流,用于广播消息
    private ServerSocket serverSocket; //accept()
    public Server5() {
        try {
            /*
                如果端口被其他程序占用了,则会发生异常:
                解决办法:
                1.更换端口号
                2.杀死占用该端口的进程(通常发生在服务端启动两次导致的情况下)
             */
            System.out.println("正在启动服务器");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务器启动完毕!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public void start(){
        /*
            ServerSocket提供的重要方法:
            Socket accept()
            用于接收客户端的连接并返回一个Socket实例与连接的客户端交互
            该方法是一个阻塞方法,调用后程序会"卡住",直到一个客户端连接为止
         */
        try {
            while (true) {
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept(); //接电话
                System.out.println("一个客户端连接上了");
                //启动一个线程负责与该客户端交互
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    public static void main(String[] args) {
        Server5 server = new Server5();
        server.start();
    }

    /**
     * 该线程任务负责与指定的客户端交互
     */
    private class ClientHandler implements Runnable{

        private Socket socket;
        private String host;//记录该客户端的IP地址

        public ClientHandler(Socket socket) {
            this.socket = socket;
            //通过socket获取远端计算机的IP地址的字符串格式
            host = socket.getInetAddress().getHostAddress();
        }

        public void run() {
            //v5:提高pw的作用域,以便在finally中可以将其从allOut删除
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream(); //低级流
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                //通过socket获取输出流用于给客户端发送消息
                OutputStream out =socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out,StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw,true);
                //V4:将对应该客户端的输出流存入到共享集合allOut中,便于其它线程可以访问到
                //v5:多线程向集合中添加元素可能产生并发安全问题,因此必须同步执行
                synchronized (allOut) {//通常同步监视器对象就选取临界资源
                    allOut.add(pw);
                }
                //v5:广播消息,该客户端上线了
                sendMessage(host+"上线了,当前在线人数"+allOut.size());

                String message;
                while ((message=br.readLine())!=null){
                    //v5:读取到客户端消息广播时改为调用sendMessage方法
                    sendMessage(host +"说"+message);
                }
            } catch (IOException e) {

            }finally {
                //将该客户端的输出流从allOut集合中删除
                synchronized (allOut) {
                    allOut.remove(pw);
                    try {
                        //关闭socket,释放资源
                        socket.close();
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
                //广播该客户端下线了
                sendMessage(host+"下线了,当前在线人数"+ allOut.size());
            }
        }

        /**
         * v5的功能一,实现广播消息的方法:
         * 将消息转发给所有客户端
         * @param message
         */
        public void sendMessage(String message){
            System.out.println(message);
            //v5:新循环就是迭代器,迭代器要求遍历过程不能用结合方法增删元素,所以需要和上面的增删互斥
            synchronized (allOut){
            for (PrintWriter o : allOut) {
                o.println(message);
            }
            }
        }
    }
}
